from ultralytics import YOLO
if __name__ == '__main__':
    model = YOLO(r'D:\Traffic_intersection\Helmet_detection\ultralytics-main\runs\detect\train2\weights\best.pt')
    model.val()
